package com.macaline.technology.redis.util;

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;

import java.io.*;
import java.util.concurrent.locks.ReentrantLock;

public class RedisTemplate {

	static ReentrantLock lock = new ReentrantLock();

	// redis服务器ip
	public static String host = "127.0.0.1";

	// redis端口
	public static int port = 6379;

	// redis密码
	public static String password = "123456";

	// 可连接实例的最大数量，默认值为8。 -1为无限制
	private static int MAX_ACTIVE = 1024;

	// 一个pool最多有多少的状态为idle（空闲）的jedis实例，默认值也是8
	private static int MAX_IDLE = 200;

	// 等到可用连接的最大时间，单位毫秒，默认值-1（永不超时）。超时会报错（JedisConnectionException）
	private static int MAX_WAIT = 4;
	public static int TIME_OUT = 3000;

	// 在borrow一个jedis实例时，是否提前进行validate操作；如果为true，则得到的jedis实例均是可用的；
	private static boolean TEST_ON_BORROW = true;

	// redis连接池
	private static JedisPool jedisPool = null;

	// 是否使用redis作为缓存
	public static boolean isUse = false;

	/**
	 * 获取redis连接池
	 *
	 * @author warm_joy
	 */
	private synchronized static JedisPool getJedisPool() {
		if (jedisPool == null) {
			JedisPoolConfig config = new JedisPoolConfig();
			config.setMaxIdle(MAX_IDLE);
			config.setMaxTotal(MAX_ACTIVE);
			config.setMaxWaitMillis(MAX_WAIT);
			config.setTestOnBorrow(TEST_ON_BORROW);
			jedisPool = new JedisPool(config, host, port, TIME_OUT);
		}
		return jedisPool;
	}

	/**
	 * 关掉连接池
	 */
	public static void shutDown(){
		getJedisPool().close();
	}


	/**
	 * 释放实例
	 *
	 * @param jedis
	 * @author warm_joy
	 */
	private static void returnResource(final Jedis jedis) {
		jedis.close();
	}

	/**
	 * 查看对象是否存在
	 *
	 * @param key 查询的key值
	 * @return
	 * @author warm_joy
	 */
	public static Boolean existObject(String key) {
		Jedis jedis = null;
		try {
			jedis = getJedisPool().getResource();
			return jedis.exists(key);
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (jedis != null) {
				jedis.close();
			}
		}
		return false;
	}

	/**
	 * 查看对象是否存在
	 *
	 * @param key 查询的key值
	 * @return
	 * @author warm_joy
	 */
	public static Boolean existObjectInBD(int BDIndex, String key) {
		Jedis jedis = null;
		try {
			jedis = getJedisPool().getResource();
			jedis.select(BDIndex);
			return jedis.exists(key);
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (jedis != null) {
				jedis.close();
			}
		}
		return false;
	}

	/**
	 * 将数据缓存到redis中
	 *
	 * @param key   缓存数据的key
	 * @param value 缓存的数据体
	 * @author warm_joy
	 */
	public static void setObject(String key, Object value) {
		Jedis jedis = null;
		try {
			jedis = getJedisPool().getResource();
			jedis.set(key.getBytes(), serialize(value));
		} catch (Exception e) {
			throw e;
		} finally {
			if (jedis != null) {
				returnResource(jedis);
			}
		}
	}

	/**
	 * 将数据缓存到redis中指定索引的数据库中
	 *
	 * @param BDIndex 进行操作的数据库索引
	 * @param key     缓存数据的key
	 * @param value   缓存的数据体
	 * @author warm_joy
	 */
	public static void selectBDsetObject(int BDIndex, String key, Object value) {
		Jedis jedis = null;
		try {
			jedis = getJedisPool().getResource();
			jedis.select(BDIndex);
			jedis.set(key.getBytes(), serialize(value));
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (jedis != null) {
				returnResource(jedis);
			}
		}
	}

	/**
	 * 将数据缓存到redis指定数据库中并设置缓存时间
	 *
	 * @param BDIndex    进行操作的数据库索引
	 * @param key        缓存数据的key
	 * @param value      缓存的数据体
	 * @param expireTime 缓存的有效时间，单位:秒(s)
	 * @author warm_joy
	 */
	public static void selectBDsetObject(int BDIndex, String key, Object value, int expireTime) {
		Jedis jedis = null;
		try {
			jedis = getJedisPool().getResource();
			jedis.select(BDIndex);
			String result = jedis.set(key.getBytes(), serialize(value));
			if (result.equals("OK")) {
				jedis.expire(key.getBytes(), expireTime);
			}
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (jedis != null) {
				returnResource(jedis);
			}
		}
	}

	/**
	 * 将数据缓存到redis中并设置缓存时间
	 *
	 * @param key        缓存数据的key
	 * @param value      缓存的数据体
	 * @param expireTime 缓存的有效时间，单位:秒(s)
	 * @author warm_joy
	 */
	public static void setObject(String key, Object value, int expireTime) {
		Jedis jedis = null;
		try {
			jedis = getJedisPool().getResource();
			String result = jedis.set(key.getBytes(), serialize(value));
			if (result.equals("OK")) {
				jedis.expire(key.getBytes(), expireTime);
			}
		} catch (Exception e) {
			throw e;
		} finally {
			if (jedis != null) {
				returnResource(jedis);
			}
		}
	}

	/**
	 * 通过key值获取redis中的缓存信息
	 *
	 * @param key 获取缓存数据的key值
	 * @return
	 * @author warm_joy
	 */
	public static Object getObject(String key) {
		Jedis jedis = null;
		try {
			jedis = getJedisPool().getResource();
			byte[] bs = jedis.get(key.getBytes());
			if (isEmpty(bs) == false) {
				return unserizlize(bs);
			}
		} catch (Exception e) {
			throw e;
		} finally {
			if (jedis != null) {
				jedis.close();
			}
		}
		return null;
	}

	/**
	 * 通过key值获取redis中的缓存信息
	 *
	 * @param BDIndex 进行操作的数据库索引
	 * @param key     获取缓存数据的key值
	 * @return
	 * @author warm_joy
	 */
	public static Object selectBDgetObject(int BDIndex, String key) {
		Jedis jedis = null;
		try {
			jedis = getJedisPool().getResource();
			jedis.select(BDIndex);
			byte[] bs = jedis.get(key.getBytes());
			if (isEmpty(bs) == false) {
				return unserizlize(bs);
			}
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (jedis != null) {
				jedis.close();
			}
		}
		return null;
	}

	/**
	 * 通过key值删除redis中的缓存值
	 *
	 * @param key 要删除缓存的key值
	 * @return
	 * @author warm_joy
	 */
	public static void delteObject(String key) {
		Jedis jedis = null;
		try {
			jedis = getJedisPool().getResource();
			jedis.del(key.getBytes());
		} catch (Exception e) {
			throw e;
		} finally {
			if (jedis != null) {
				jedis.close();
			}
		}
	}

	/**
	 * 通过key值删除redis中的缓存值
	 *
	 * @param BDIndex 进行操作的数据库索引
	 * @param key     要删除缓存的key值
	 * @return
	 * @author warm_joy
	 */
	public static void selectBDdelteObject(int BDIndex, String key) {
		Jedis jedis = null;
		try {
			jedis = getJedisPool().getResource();
			jedis.select(BDIndex);
			jedis.del(key.getBytes());
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (jedis != null) {
				jedis.close();
			}
		}
	}

	/**
	 * 查验查询出的数据是否为空
	 *
	 * @param bs 需要检验的数据
	 * @return true：数据异常或者数据为空；false：数据不为空；
	 * @author warm_joy
	 */
	private static boolean isEmpty(byte[] bs) {
		return (bs == null || bs.length == 0);
	}

	/**
	 * 将要缓存的对象进行序列化处理
	 *
	 * @param obj 需要序列化的对象
	 * @author warm_joy
	 */
	public static byte[] serialize(Object obj) {
		ObjectOutputStream obi = null;
		ByteArrayOutputStream bai = null;
		try {
			bai = new ByteArrayOutputStream();
			obi = new ObjectOutputStream(bai);
			obi.writeObject(obj);
			byte[] byt = bai.toByteArray();
			return byt;
		} catch (IOException e) {
			e.printStackTrace();
		}
		return null;
	}

	/**
	 * 对字节数据数据进行反序列化编译
	 *
	 * @param byt 需要编译的字节数组
	 * @author warm_joy
	 */
	public static Object unserizlize(byte[] byt) {
		ObjectInputStream oii = null;
		ByteArrayInputStream bis = null;
		bis = new ByteArrayInputStream(byt);
		try {
			oii = new ObjectInputStream(bis);
			Object obj = oii.readObject();
			return obj;
		} catch (Exception e) {
			e.printStackTrace();
		}
		return null;
	}
}
